import unittest
import TestCase_01

suite = unittest.TestSuite()

# 单个导入
# suite.addTest(TestCase_01.TestDemo("test_method1"))
# suite.addTest(TestCase_01.TestDemo("test_method2"))

# 添加整个测试类
suite.addTest(unittest.makeSuite(TestCase_01.TestDemo))


runner = unittest.TextTestRunner()

runner.run(suite)
